// 计算字符串的 SHA256 哈希值

package utils

import (
	"crypto/md5"
	"crypto/sha256"
	"encoding/hex"
	"fmt"
	"io"
	"os"
)

type XinmanHasher struct{}

func (xinmanHasher XinmanHasher) Sha256(str string) string {
	h := sha256.New()
	h.Write([]byte(str))
	return hex.EncodeToString(h.Sum(nil))
}

// 计算文件的SHA256 哈希值
func (xinmanHasher XinmanHasher) Sha256File(filePath string) (string, error) {
	file, err := os.Open(filePath)
	if err != nil {
		return "", fmt.Errorf("could not open file: %v", err)
	}
	defer file.Close()

	// 创建一个 SHA-256 哈希计算器
	hash := sha256.New()

	// 将文件内容写入到 hash 计算器中
	_, err = io.Copy(hash, file)
	if err != nil {
		return "", fmt.Errorf("could not read file: %v", err)
	}

	// 计算文件的 SHA-256 哈希值
	hashInBytes := hash.Sum(nil)

	// 将结果转换为十六进制字符串
	return fmt.Sprintf("%x", hashInBytes), nil

}

// 计算MD5
func (xinmanHasher XinmanHasher) Md5(str string) string {
	h := md5.New()
	h.Write([]byte(str))
	return hex.EncodeToString(h.Sum(nil))
}

// 计算文件的MD5
func (xinmanHasher XinmanHasher) Md5File(filePath string) (string, error) {
	file, err := os.Open(filePath)
	if err != nil {
		return "", fmt.Errorf("could not open file: %v", err)
	}
	defer file.Close()

	// 创建一个 MD5 哈希计算器
	hash := md5.New()

	// 将文件内容写入到 hash 计算器中
	_, err = io.Copy(hash, file)
	if err != nil {
		return "", fmt.Errorf("could not read file: %v", err)
	}

	// 计算文件的 MD5 哈希值
	hashInBytes := hash.Sum(nil)

	// 将结果转换为十六进制字符串
	return fmt.Sprintf("%x", hashInBytes), nil

}
